#!/bin/bash
# TransUNet训练启动脚本
export CUDA_VISIBLE_DEVICES=0

conda init

source "/data/apps/miniforge3/etc/profile.d/conda.sh"

conda activate transunet_py37

PROJECT_ROOT="./"

# 启动训练任务
python ${PROJECT_ROOT}/DA-TransUNet/train.py \
    --dataset crc \
    --vit_name R50-ViT-B_16 \
    --img_size 224 \
    --vit_patches_size 16 \
    --n_skip 3 \
    --max_epochs 50 \
    --batch_size 16 \
    --base_lr 0.01 \
    --seed 2025

# 可选：训练完成后执行的操作
echo "Training completed!"